package org.deng.learn.concurrent.basic;

import java.util.concurrent.CountDownLatch;

/**
 * @author 邓力宾
 * @date 2020/5/15 下午4:15
 * @desc
 *  CountDownLatch是通过一个计数器来实现的，计数器的初始值为线程的数量。每当一个线程完成了自己的任务后，计数器的值就会减1。
 *   当计数器值到达0时，它表示所有的线程已经完成了任务，然后在闭锁上等待的线程就可以恢复执行任务。
 *  构造器中的计数值（count）实际上就是闭锁需要等待的线程数量。这个值只能被设置一次，而且CountDownLatch没有提供任何机制去重新设置这个计数值。
 *
 *  与CountDownLatch的第一次交互是主线程等待其他线程。主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。
 *  这样主线程的操作就会在这个方法上阻塞，直到其他线程完成各自的任务。
 */
public class CountDownLatchDemo {
    /**
     * 假设一条流水线上有三个工作者：worker0，worker1，worker2。有一个任务的完成需要他们三者协作完成，
     * worker2可以开始这个任务的前提是worker0和worker1完成了他们的工作，而worker0和worker1是可以并行他们各自的工作的。
     */
    static class Worker implements Runnable {

        private String name;
        private CountDownLatch countDownLatch;

        public Worker(String name, CountDownLatch countDownLatch) {
            this.name = name;
            this.countDownLatch = countDownLatch;
        }

        @Override
        public void run() {
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
            System.out.println(name + "完成工作");
            countDownLatch.countDown();
            //调用一次这个方法，在构造函数中初始化的count值就减1。所以当N个线程都调 用了这个方法，
            // count的值等于0，然后主线程就能通过await()方法，恢复执行自己的任务

        }
    }
    public static void main(String[] args) throws InterruptedException {
        CountDownLatch countDownLatch = new CountDownLatch(2);
        Worker worker1 = new Worker("工人1",countDownLatch);
        Worker worker2 = new Worker("工人2",countDownLatch);
        Worker worker3 = new Worker("工人3",countDownLatch);

        Thread thread1 = new Thread(worker1);
        Thread thread2 = new Thread(worker2);
        Thread thread3 = new Thread(worker3);

        System.out.println("工人1开始工作");
        thread1.start();
        System.out.println("工人2开始工作");
        thread2.start();
        countDownLatch.await();

        System.out.println("工人3开始工作");
        thread3.start();
    }
}
